package com.alibaba.datax.plugin.reader.ossexcelreader;

import io.minio.*;
import io.minio.messages.Bucket;
import io.minio.messages.Item;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import sun.misc.BASE64Decoder;

import java.io.ByteArrayInputStream;
import java.io.InputStream;
import java.io.UnsupportedEncodingException;
import java.net.URLDecoder;
import java.util.HashSet;
import java.util.List;
import java.util.Optional;
import java.util.Set;

/**
 * MinIO工具类
 *
 */
public class MinioUtils {

    private static final Logger LOG = LoggerFactory.getLogger(MinioUtils.class);

    private MinioClient minioClient;

    public MinioUtils(MinioClient minioClient) {
        this.minioClient = minioClient;
    }

    /**
     * 判断Bucket是否存在，true：存在，false：不存在
     *
     * @param bucketName 存储桶
     * @return T/F
     */
    public boolean bucketExists(String bucketName) throws Exception{
        return minioClient.bucketExists(BucketExistsArgs.builder().bucket(bucketName).build());
    }


    /**
     * 获得所有Bucket列表
     *
     * @return list
     */
    public List<Bucket> getAllBuckets() throws Exception{
        return minioClient.listBuckets();
    }

    /**
     * 根据bucketName获取其相关信息
     *
     * @param bucketName 存储桶
     * @return obj
     */
    public Optional<Bucket> getBucket(String bucketName) throws Exception{
        return getAllBuckets().stream().filter(b -> b.name().equals(bucketName)).findFirst();
    }

    /**
     * 判断文件是否存在
     *
     * @param bucketName 存储桶
     * @param objectName 文件名称
     * @return T/F
     */
    public boolean isObjectExist(String bucketName, String objectName) {
        boolean exist = true;
        try {
            minioClient.statObject(StatObjectArgs.builder().bucket(bucketName).object(objectName).build());
        } catch (Exception e) {
            LOG.error("[Minio工具类]>>>> 判断文件是否存在, 异常：", e);
            exist = false;
        }
        return exist;
    }

    /**
     * 判断文件夹是否存在
     *
     * @param bucketName 存储桶
     * @param objectName 文件夹名称
     * @return T/F
     */
    public boolean isFolderExist(String bucketName, String objectName) {
        boolean exist = false;
        try {
            Iterable<Result<Item>> results = minioClient.listObjects(
                    ListObjectsArgs.builder().bucket(bucketName).prefix(objectName.substring(0, objectName.length() -1)).recursive(false).build());
            for (Result<Item> result : results) {
                Item item = result.get();
                if (item.isDir() && objectName.equals(item.objectName())) {
                    exist = true;
                }
            }
        } catch (Exception e) {
            LOG.error("[Minio工具类]>>>> 判断文件夹是否存在，异常：", e);
            exist = false;
        }
        return exist;
    }


    /**
     * 获取文件流
     *
     * @param bucketName 存储桶
     * @param objectName 文件名
     * @return 二进制流
     */
    public InputStream getObject(String bucketName, String objectName) throws Exception {
        return minioClient.getObject(
                GetObjectArgs.builder()
                        .bucket(bucketName)
                        .object(objectName)
                        .build());
    }


    /**
     * 获取路径下文件列表
     *
     * @param bucketName 存储桶
     * @param prefix     文件名称
     * @param recursive  是否递归查找，false：模拟文件夹结构查找
     * @return 二进制流
     */
    public Iterable<Result<Item>> listObjects(String bucketName, String prefix, boolean recursive) {
        return minioClient.listObjects(
                ListObjectsArgs.builder()
                        .bucket(bucketName)
                        .prefix(prefix)
                        .recursive(recursive)
                        .build());
    }

    /**
     * 获取路径下的文件列表
     * @param bucketName 存储桶
     * @param prefix 文件名称
     * @return path
     */
    public Set<String> listObjects(String bucketName, String prefix) {
        Set<String> set = new HashSet<>();
        try {
            Iterable<Result<Item>> results = this.listObjects(bucketName, prefix, false);
            for (Result<Item> result : results) {
                Item item = result.get();
                if (verifyFileSuffix(item.objectName())) {
                    set.add(item.objectName());
                }
            }
        } catch (Exception e) {
            LOG.error("[Minio工具类]>>>> 获取文件夹下的文件失败，异常：", e);
        }
        return set;
    }

    /**
     * 校验文件须以xls,xlsx结尾
     * @param fileName 文件名称
     * @return T：校验不通过
     */
    private boolean verifyFileSuffix(String fileName) {
        if (fileName.endsWith(Constant.XLS_SUFFIX) || fileName.endsWith(Constant.XLSX_SUFFIX)) {
            return true;
        }
        return false;
    }


    public static InputStream base64ToInputStream(String base64) {
        ByteArrayInputStream stream = null;
        try {
            byte[] bytes = new BASE64Decoder().decodeBuffer(base64.trim());
            stream = new ByteArrayInputStream(bytes);
        } catch (Exception e) {
            e.printStackTrace();
        }
        return stream;
    }


    /**
     * 将URLDecoder编码转成UTF8
     *
     * @param str url
     * @return str
     */
    public String getUtf8ByURLDecoder(String str) throws UnsupportedEncodingException {
        String url = str.replaceAll("%(?![0-9a-fA-F]{2})", "%25");
        return URLDecoder.decode(url, "UTF-8");
    }
}